
"""
当前目录下的模板
"""

from core.template_parser import TemplateParser
from pathlib import Path


def get_data_analyze_prompts():

    current_dir = Path(__file__).resolve().parent

    md_files = ["task_rewrite.md", "fetch_data.md", "task_decompose.md", "report.md"]

    template_prompts = dict()
    for filename in md_files:
        file = current_dir / filename
        with open(current_dir / filename, 'r', encoding='utf-8') as fs:
            content = fs.read()
        template = TemplateParser(content)
        template_prompts[file.stem] = template
    return template_prompts

data_analyze_prompts = get_data_analyze_prompts()